<?php

namespace app\admin\library\traits;

use app\admin\library\Auth;
use Exception;
use PhpOffice\PhpSpreadsheet\Cell\Coordinate;
use PhpOffice\PhpSpreadsheet\Reader\Xlsx;
use PhpOffice\PhpSpreadsheet\Reader\Xls;
use PhpOffice\PhpSpreadsheet\Reader\Csv;
use think\db\exception\BindParamException;
use think\exception\DbException;
use think\exception\PDOException;
use think\response\Json;

trait Backend
{
    /** (only post)从输入的ids中根据用户权限过滤出允许操作的ids
     * @param $ids
     * @return bool|\PDOStatement|string|\think\Collection
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\ModelNotFoundException
     * @throws \think\exception\DbException
     */
    protected function authIds($ids, $noRowErrorInfo = 'no row', $onlyTrashed = false)
    {
        $this->tryPost();
        $ids = $ids ?? $this->post($ids);

        $pk = $this->model->getPk();
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds)) {
            $this->model->where($this->dataLimitField, 'in', $adminIds);
        }
        if (!$onlyTrashed) {
            $list = $this->model->where($pk, 'in', $ids)->select();
        } else {
            $list = $this->model->onlyTrashed()->where($pk, 'in', $ids)->select();
        }

        if (count($list) <= 0) {
            ajaxerror($noRowErrorInfo);
        }

        return $list;
    }
    /**
     * 排除前台提交过来的字段
     * @param $params
     * @return array
     */
    protected function preExcludeFields($params)
    {
        if (is_array($this->excludeFields)) {
            foreach ($this->excludeFields as $field) {
                if (array_key_exists($field, $params)) {
                    unset($params[$field]);
                }
            }
        } else if (array_key_exists($this->excludeFields, $params)) {
            unset($params[$this->excludeFields]);
        }
        return $params;
    }

    /**
     * 查看
     *
     * @return string|Json
     * @throws \think\Exception
     * @throws DbException
     */
    public function index()
    {
        //设置过滤方法
        $this->request->filter(['strip_tags', 'trim']);
        if (false === $this->request->isAjax()) {
            return $this->view->fetch();
        }
        //如果发送的来源是 Selectpage，则转发到 Selectpage
        if ($this->request->request('keyField')) {
            return $this->selectpage();
        }
        [$where, $sort, $order, $offset, $limit] = $this->buildparams();
        $list = $this->model
            ->where($where)
            ->order($sort, $order)
            ->paginate($limit);
        $result = ['total' => $list->total(), 'rows' => $list->items()];
        return json($result);
    }

    /**
     * 回收站
     *
     * @return string|Json
     * @throws \think\Exception
     */
    public function recyclebin()
    {
        //设置过滤方法
        $this->request->filter(['strip_tags', 'trim']);
        if (false === $this->request->isAjax()) {
            return $this->view->fetch();
        }
        [$where, $sort, $order, $offset, $limit] = $this->buildparams();
        $list = $this->model
            ->onlyTrashed()
            ->where($where)
            ->order($sort, $order)
            ->paginate($limit);
        $result = ['total' => $list->total(), 'rows' => $list->items()];
        return json($result);
    }

    // 检查并获取数据model
    protected function row($ids)
    {
        $row = $this->model->get($ids);
        if (!$row) {
            ajaxerror(__('No Results were found'));
        }
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds)) {
            if (!in_array($row[$this->dataLimitField], $adminIds)) {
                ajaxerror(__('You have no permission'));
            }
        }

        return $row;
    }

    /**
     * 添加
     */
    public function add()
    {
        if (!$this->request->isPost()) {
            return $this->show_add();
        }

        $params = $this->request->post($this->data_tig);
        if (!$params) {
            $this->error(__('Parameter %s can not be empty', ''));
        }
        __db(
            function () use ($params) {

                $params = __check($this->before_add($params), __("Internal error for process parameters"));

                // bug修复
                // 如果表单已填写dataLimitField字段，则不进行自动填充
                if ($this->dataLimit && $this->dataLimitFieldAutoFill && !isset($params[$this->dataLimitField])) {
                    if ($this->dataLimitField == 'industry_id') {
                        $params[$this->dataLimitField] = $this->industry_id;
                    } else {
                        $params[$this->dataLimitField] = $this->auth->id;
                    }
                }

                //是否采用模型验证
                if ($this->modelValidate) {
                    $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                    $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.add' : $name) : $this->modelValidate;
                    $this->model->validateFailException(true)->validate($validate);
                }
                $this->model->allowField(true)->save($params);

                $this->after_add($this->model->id, $params);
                return true;
            },
            null,
            true
        );

        $this->success();
    }

    /**
     * 添加，查看视图
     */
    protected function show_add()
    {
        return $this->view->fetch();
    }

    /**
     * 添加前回调函数
     *
     * @param array $params
     * @return array new_params
     */
    protected function before_add($params)
    {
        return $params;
    }

    /**
     * 添加后回调函数
     *
     * @param int|string $newid
     * @param array $params
     */
    protected function after_add($newid, $params)
    {
    }

    /**
     * 编辑
     */
    public function edit($ids = null)
    {
        $row = $this->row($ids);
        if (!$this->request->isPost()) {
            return $this->show_edit($row);
        }

        $params = $this->request->post($this->data_tig);
        if (!$params) {
            ajaxerror(__('Parameter %s can not be empty', ''));
        }
        __db(
            function () use ($params, $row) {

                $originData = $row->getData();
                $changeData = array_diff_values($params, $row->getData());
                //            $overData = array_merge($row->getData(), $params);

                $params = __check($this->before_edit($params, $changeData, $originData), __("No data were updated"));

                //是否采用模型验证
                if ($this->modelValidate) {
                    $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                    $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.edit' : $name) : $this->modelValidate;
                    $row->validateFailException(true)->validate($validate);
                }
                $result = $row->allowField(true)->save($params);
                $this->after_edit($row, $changeData, $originData);
                return $result;

            },
            null,
            true
        );

        $this->success();
    }

    protected function show_edit($row)
    {
        $this->view->assign("row", $row);
        return $this->view->fetch();
    }

    /**
     * 编辑前回调函数
     *
     * @param array params
     * @return array new_params
     */
    protected function before_edit($params, $changeData, $originData)
    {
        return $params;
    }

    /**
     * 编辑后回调函数
     *
     * @param array row
     */
    protected function after_edit($row, $changeData, $originData)
    {
    }

    /**
     * 删除
     */

    protected function before_del($ids) {
        return $ids;
    }

    public function del($ids = "")
    {
        $ids = __check($this->before_del($ids), __("No data were updated"));
        $list = $this->authIds($ids, 'No rows were deleted');
        __db(
            function () use ($list) {
                foreach ($list as $k => $v) {
                    $v->delete();
                    $this->after_del($v);
                }
            }
        );

        ajaxsuccess();
    }

    protected function after_del($v)
    {
        return;
    }

    /**
     * 真实删除
     */
    public function destroy($ids = "")
    {
        $list = $this->authIds($ids, 'No rows were deleted', true);
        __db(
            function () use ($list) {
                foreach ($list as $k => $v) {
                    $v->delete();
                    $this->after_destroy($v);
                }
            }
        );

        ajaxsuccess();
    }

    protected function after_destroy($v)
    {
        return;
    }

    /**
     * 还原
     */
    public function restore($ids = "")
    {
        $list = $this->authIds($ids, 'No rows were updated', true);
        __db(
            function () use ($list) {
                foreach ($list as $k => $v) {
                    $v->restore();
                    $this->after_restore($v);
                }
            }
        );

        ajaxsuccess();
    }

    protected function after_restore($v)
    {
        return;
    }

    /**
     * 批量更新
     */
    public function multi($ids = "")
    {
        parse_str($this->post("params"), $values);
        $values = $this->auth->isSuperAdmin() ? $values : array_intersect_key($values, array_flip(is_array($this->multiFields) ? $this->multiFields : explode(',', $this->multiFields)));
        __check($values);

        $list = $this->authIds($ids, 'No rows were updated');
        __db(
            function () use ($list, $values) {
                foreach ($list as $k => $v) {
                    $v->allowField(true)->isUpdate(true)->save($values);
                    $this->after_multi($values, $v);
                }
            }
        );

        ajaxsuccess();
    }

    protected function after_multi($values, $v)
    {
        return;
    }

    /**
     * 导入
     *
     * @return void
     * @throws PDOException
     * @throws BindParamException
     */
    protected function import()
    {
        $file = $this->request->request('file');
        if (!$file) {
            $this->error(__('Parameter %s can not be empty', 'file'));
        }
        $filePath = ROOT_PATH . DS . 'public' . DS . $file;
        if (!is_file($filePath)) {
            $this->error(__('No results were found'));
        }
        //实例化reader
        $ext = pathinfo($filePath, PATHINFO_EXTENSION);
        if (!in_array($ext, ['csv', 'xls', 'xlsx'])) {
            $this->error(__('Unknown data format'));
        }
        if ($ext === 'csv') {
            $file = fopen($filePath, 'r');
            $filePath = tempnam(sys_get_temp_dir(), 'import_csv');
            $fp = fopen($filePath, 'w');
            $n = 0;
            while ($line = fgets($file)) {
                $line = rtrim($line, "\n\r\0");
                $encoding = mb_detect_encoding($line, ['utf-8', 'gbk', 'latin1', 'big5']);
                if ($encoding !== 'utf-8') {
                    $line = mb_convert_encoding($line, 'utf-8', $encoding);
                }
                if ($n == 0 || preg_match('/^".*"$/', $line)) {
                    fwrite($fp, $line . "\n");
                } else {
                    fwrite($fp, '"' . str_replace(['"', ','], ['""', '","'], $line) . "\"\n");
                }
                $n++;
            }
            fclose($file) || fclose($fp);

            $reader = new Csv();
        } elseif ($ext === 'xls') {
            $reader = new Xls();
        } else {
            $reader = new Xlsx();
        }

        //导入文件首行类型,默认是注释,如果需要使用字段名称请使用name
        $importHeadType = isset($this->importHeadType) ? $this->importHeadType : 'comment';

        $table = $this->model->getQuery()->getTable();
        $database = \think\Config::get('database.database');
        $fieldArr = [];
        $list = db()->query("SELECT COLUMN_NAME,COLUMN_COMMENT FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = ? AND TABLE_SCHEMA = ?", [$table, $database]);
        foreach ($list as $k => $v) {
            if ($importHeadType == 'comment') {
                $v['COLUMN_COMMENT'] = explode(':', $v['COLUMN_COMMENT'])[0]; //字段备注有:时截取
                $fieldArr[$v['COLUMN_COMMENT']] = $v['COLUMN_NAME'];
            } else {
                $fieldArr[$v['COLUMN_NAME']] = $v['COLUMN_NAME'];
            }
        }

        //加载文件
        $insert = [];
        try {
            if (!$PHPExcel = $reader->load($filePath)) {
                $this->error(__('Unknown data format'));
            }
            $currentSheet = $PHPExcel->getSheet(0);  //读取文件中的第一个工作表
            $allColumn = $currentSheet->getHighestDataColumn(); //取得最大的列号
            $allRow = $currentSheet->getHighestRow(); //取得一共有多少行
            $maxColumnNumber = Coordinate::columnIndexFromString($allColumn);
            $fields = [];
            for ($currentRow = 1; $currentRow <= 1; $currentRow++) {
                for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
                    $val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getValue();
                    $fields[] = $val;
                }
            }

            for ($currentRow = 2; $currentRow <= $allRow; $currentRow++) {
                $values = [];
                for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
                    $val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getValue();
                    $values[] = is_null($val) ? '' : $val;
                }
                $row = [];
                $temp = array_combine($fields, $values);
                foreach ($temp as $k => $v) {
                    if (isset($fieldArr[$k]) && $k !== '') {
                        $row[$fieldArr[$k]] = $v;
                    }
                }
                if ($row) {
                    $insert[] = $row;
                }
            }
        } catch (Exception $exception) {
            $this->error($exception->getMessage());
        }
        if (!$insert) {
            $this->error(__('No rows were updated'));
        }

        try {
            //是否包含admin_id字段
            $has_admin_id = false;
            foreach ($fieldArr as $name => $key) {
                if ($key == 'admin_id') {
                    $has_admin_id = true;
                    break;
                }
            }
            if ($has_admin_id) {
                $auth = Auth::instance();
                foreach ($insert as &$val) {
                    if (!isset($val['admin_id']) || empty($val['admin_id'])) {
                        $val['admin_id'] = $auth->isLogin() ? $auth->id : 0;
                    }
                }
            }
            $this->model->saveAll($insert);
        } catch (PDOException $exception) {
            $msg = $exception->getMessage();
            if (preg_match("/.+Integrity constraint violation: 1062 Duplicate entry '(.+)' for key '(.+)'/is", $msg, $matches)) {
                $msg = "导入失败，包含【{$matches[1]}】的记录已存在";
            };
            $this->error($msg);
        } catch (Exception $e) {
            $this->error($e->getMessage());
        }

        $this->success();
    }
}
